import torch
from Cbam import Cbam
from SENet import SENet
from ECA import ECANet
from Cbam_ECA import CbamECA
from darknet53 import darknet53

model = Cbam(512, 16, 7)
inputs = torch.randn(2, 512, 26, 26)
print(model(inputs).shape)

model = SENet(512)
inputs = torch.randn(2, 512, 26, 26)
print(model(inputs).shape)

model = ECANet(512)
inputs = torch.randn(2, 512, 26, 26)
print(model(inputs).shape)

model = CbamECA(512)
inputs = torch.randn(2, 512, 26, 26)
print(model(inputs).shape)

# model = darknet53()
# inputs = torch.randn(2, 3, 1024, 768)
# print(model(inputs)[0].shape)
# print(model(inputs)[1].shape)
# """
# torch.Size([2, 256, 128, 96])
# torch.Size([2, 512, 64, 48])
# """
#
